import argparse

class InputParser:
    def __init__(self):
        # set defaults
        self.default_number_runs = 100
        self.default_seed = 42

    def parse_input(self):
        parser = argparse.ArgumentParser()
        # parser.add_argument("dimension", 
        #                     choices=[1],
        #                     help="dimension of Euclidian space",
        #                     type=int)
        parser.add_argument("number_vertices",
                            help="number of vertices in graph",
                            type=int)
        parser.add_argument("-nr", "--number_runs",
                            help="number of runs",
                            type=int)
        parser.add_argument("-s", "--seed",
                            help="value for initialization RNG",
                            type=int)
        args = parser.parse_args()
        if not args.number_runs:
            args.number_runs = self.default_number_runs
        if not args.seed:
            args.seed = self.default_seed
        return args
